//
//  75_颜色分类.swift
//  Swift-LeetCode
//
//  Created by 卢悦明 on 2024/3/8.
/**
 给定一个包含红色、白色和蓝色、共 n 个元素的数组 nums ，原地对它们进行排序，使得相同颜色的元素相邻，并按照红色、白色、蓝色顺序排列。

 我们使用整数 0、 1 和 2 分别表示红色、白色和蓝色。

 必须在不使用库内置的 sort 函数的情况下解决这个问题。

  

 示例 1：

 输入：nums = [2,0,2,1,1,0]
 输出：[0,0,1,1,2,2]
 示例 2：

 输入：nums = [2,0,1]
 输出：[0,1,2]
 */

import UIKit

class HeLanGuoQi: NSObject {
    func QA() {
        var array = [2,0,1]
        sortColors(&array)
        print(array)
    }
    // 空间复杂度 O(1) 时间复杂度O(n) 数据只有0， 1， 2三个数（荷兰国旗问题）
    func sortColors(_ nums: inout [Int]) {
        var begin = 0
        var end = nums.count - 1
        var current = 0
        while current <= end {
            let num = nums[current]
            if num == 0 {
                (nums[begin], nums[current]) =  (nums[current], nums[begin])
                begin += 1
                current += 1
            } else if num == 2 { // = 2要交换位置，然后对当前current再判断一下（current == 2 && end == 2）
                (nums[end], nums[current]) =  (nums[current], nums[end])
                end -= 1
            } else { // 等于1 的情况继续扫描
                current += 1
            }
        }
    }
}
